package org.jeecg.common.datasource.aop;

import org.aspectj.lang.annotation.Aspect;
import org.springframework.aop.Advisor;
import org.springframework.aop.aspectj.AspectJExpressionPointcut;
import org.springframework.aop.support.DefaultPointcutAdvisor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.transaction.TransactionDefinition;
import org.springframework.transaction.TransactionManager;
import org.springframework.transaction.interceptor.*;

import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

/**
 * @description 通过AOP切面设置全局事务，拦截service包下面所有方法
 */
@Configuration
@Aspect
public class TransactionManagerAspect {
    // 单位秒，默认10分钟
    private static final int TX_METHOD_TIMEOUT = 600;
    /**
     * 定义切点变量：拦截service包下所有类的所有方法,返回值类型任意的方法
     **/
    private static final String AOP_POINTCUT_EXPRESSION = "execution (* org.jeecg..*.*.service..*(..))";

    @Autowired
    private TransactionManager transactionManager;

    /**
     * @author snow
     * @data 2021年8月24日
     * @description springBoot事务配置
     */
    public TransactionInterceptor TxAdvice() {
        Map<String, TransactionAttribute> txMap = new HashMap<>();
        /* 事务管理规则，声明具备事务管理的方法名 */
        NameMatchTransactionAttributeSource source = new NameMatchTransactionAttributeSource();
        /* 只读事物、不做更新删除等 */
        /* 当前存在事务就用当前的事务，当前不存在事务就创建一个新的事务 */
        RuleBasedTransactionAttribute readOnlyRule = new RuleBasedTransactionAttribute();
        /* 设置当前事务是否为只读事务，true为只读 */
        readOnlyRule.setReadOnly(true);
        /* PROPAGATION_SUPPORTS 如果当前存在事务，则加入该事务；如果当前没有事务，则以非事务的方式继续运行 */
        readOnlyRule.setPropagationBehavior(TransactionDefinition.PROPAGATION_SUPPORTS);
        txMap.put("get*", readOnlyRule);
        txMap.put("query*", readOnlyRule);
        txMap.put("find*", readOnlyRule);
        txMap.put("select*", readOnlyRule);
        txMap.put("search*", readOnlyRule);


        RuleBasedTransactionAttribute requireRule = new RuleBasedTransactionAttribute();
        /* 抛出异常后执行切点回滚 */
        requireRule.setRollbackRules(Collections.singletonList(new RollbackRuleAttribute(Exception.class)));
        /* PROPAGATION_REQUIRED:事务隔离性为1，若当前存在事务，则加入该事务；如果当前没有事务，则创建一个新的事务。这是默认值。*/
        requireRule.setPropagationBehavior(TransactionDefinition.PROPAGATION_REQUIRED);
        /* 设置事务失效时间，如果超过10分钟，则回滚事务 */
        requireRule.setTimeout(TX_METHOD_TIMEOUT);

        txMap.put("add*", requireRule);
        txMap.put("save*", requireRule);
        txMap.put("insert*", requireRule);
        txMap.put("update*", requireRule);
        txMap.put("delete*", requireRule);
        txMap.put("remove*", requireRule);
        txMap.put("import*", requireRule);
        txMap.put("sync*", requireRule);
        txMap.put("copy*", requireRule);
        txMap.put("create*", requireRule);
        txMap.put("batch*", requireRule);
        txMap.put("exec*", requireRule);
        txMap.put("copy*", requireRule);
        txMap.put("set*", requireRule);
        txMap.put("change*", requireRule);
        txMap.put("move*", requireRule);
        txMap.put("clear*", requireRule);

        /* 当前存在事务就用当前的事务，当前不存在事务就创建一个新的事务 */
        RuleBasedTransactionAttribute supportRule = new RuleBasedTransactionAttribute();
        /* PROPAGATION_SUPPORTS 如果当前存在事务，则加入该事务；如果当前没有事务，则以非事务的方式继续运行 */
        supportRule.setPropagationBehavior(TransactionDefinition.PROPAGATION_SUPPORTS);
        txMap.put("*", supportRule);
        source.setNameMap(txMap);

        TransactionInterceptor txAdvice = new TransactionInterceptor(transactionManager, source);
        return txAdvice;
    }

    @Bean
    public Advisor txAdviceAdvisor() {
        /* 声明切点的面
         * 切面（Aspect）：切面就是通知和切入点的结合。通知和切入点共同定义了关于切面的全部内容——它的功能、在何时和何地完成其功能。
         * */
        AspectJExpressionPointcut pointcut = new AspectJExpressionPointcut();
        /*声明和设置需要拦截的方法,用切点语言描写*/
        pointcut.setExpression(AOP_POINTCUT_EXPRESSION);
        /*设置切面=切点pointcut+通知TxAdvice*/
        return new DefaultPointcutAdvisor(pointcut, this.TxAdvice());
    }
}
